import numpy as np

def postprocess_output(output):
    # 假设output是模型输出的概率向量，这里简单取概率最大的索引作为预测类别（示例，按实际改）
    predicted_class = np.argmax(output.detach().numpy(), axis=1)[0]
    return predicted_class